package org.dromara.common.mybatis.helper;

import cn.dev33.satoken.context.SaHolder;
import cn.dev33.satoken.context.model.SaStorage;
import cn.hutool.core.util.ObjectUtil;
import com.baomidou.mybatisplus.core.plugins.IgnoreStrategy;
import com.baomidou.mybatisplus.core.plugins.InterceptorIgnoreHelper;
import lombok.AccessLevel;
import lombok.NoArgsConstructor;

import java.util.HashMap;
import java.util.Map;
import java.util.function.Supplier;


/**
 * @Author 刘武贵
 * @Date 2024/8/15 22:00
 * @Description 功能描述：数据权限助手
 */
@NoArgsConstructor(access = AccessLevel.PRIVATE)
@SuppressWarnings("unchecked cast")
public class DataPermissionHelper {

    /**
     * 定义一个常量，用于标识数据权限的键名
     * 该键名用于在数据权限检查中，标识需要检查的数据权限对应的键
     */
    private static final String DATA_PERMISSION_KEY = "data:permission";

    /**
     * 从上下文中获取指定键的变量值，并将其转换为指定的类型
     *
     * @param key 变量的键
     * @param <T> 变量值的类型
     * @return 指定键的变量值，如果不存在则返回 null
     */
    public static <T> T getVariable(String key) {
        Map<String, Object> context = getContext();
        return (T) context.get(key);
    }

    /**
     * 向上下文中设置指定键的变量值
     *
     * @param key   要设置的变量的键
     * @param value 要设置的变量值
     */
    public static void setVariable(String key, Object value) {
        Map<String, Object> context = getContext();
        context.put(key, value);
    }

    /**
     * 获取当前线程的数据权限上下文
     *
     * 此方法通过从当前线程的存储中检索特定键（DATA_PERMISSION_KEY）来获取数据权限信息
     * 如果未找到数据权限信息，则会创建一个新的HashMap，并将其存储回线程的存储中
     * 这样可以确保每次调用此方法时，都能获得一个非空的Map对象
     *
     * @return 存储在SaStorage中的Map对象，用于存储数据权限相关的上下文信息
     * @throws NullPointerException 如果数据权限上下文类型异常，则抛出NullPointerException
     */
    public static Map<String, Object> getContext() {
        // 获取当前线程的存储
        SaStorage saStorage = SaHolder.getStorage();
        // 尝试从线程存储中获取数据权限信息
        Object attribute = saStorage.get(DATA_PERMISSION_KEY);
        // 如果数据权限信息不存在，则创建一个空的HashMap，并存储回线程存储中
        if (ObjectUtil.isNull(attribute)) {
            saStorage.set(DATA_PERMISSION_KEY, new HashMap<>());
            attribute = saStorage.get(DATA_PERMISSION_KEY);
        }
        // 检查获取的数据权限信息是否为Map类型
        if (attribute instanceof Map map) {
            // 如果是Map类型，则返回该Map对象
            return map;
        }
        throw new NullPointerException("data permission context type exception");
    }

    /**
     * 开启忽略数据权限(开启后需手动调用 {@link #disableIgnore()} 关闭)
     */
    public static void enableIgnore() {
        // 通过构建一个忽略策略，设置数据权限为true，来实现忽略数据权限检查的功能
        InterceptorIgnoreHelper.handle(IgnoreStrategy.builder().dataPermission(true).build());
    }

    /**
     * 关闭忽略数据权限
     */
    public static void disableIgnore() {
        // 清除已设置的忽略策略，确保数据权限检查能够正常进行
        InterceptorIgnoreHelper.clearIgnoreStrategy();
    }

    /**
     * 在忽略数据权限中执行
     * <p>禁止在忽略数据权限中执行忽略数据权限</p>
     *
     * @param handle 处理执行方法
     */
    public static void ignore(Runnable handle) {
        // 开启忽略数据权限(开启后需手动调用 {@link #disableIgnore()} 关闭)
        enableIgnore();
        try {
            handle.run();
        } finally {
            // 关闭忽略数据权限
            disableIgnore();
        }
    }

    /**
     * 在忽略数据权限中执行
     * <p>禁止在忽略数据权限中执行忽略数据权限</p>
     *
     * @param handle 处理执行方法
     */
    public static <T> T ignore(Supplier<T> handle) {
        // 开启忽略数据权限(开启后需手动调用 {@link #disableIgnore()} 关闭)
        enableIgnore();
        try {
            return handle.get();
        } finally {
            // 关闭忽略数据权限
            disableIgnore();
        }
    }

}
